


set -e

set -x 



# poetry run  ./scripts/train.sh --config configs/train/avid/avid_145M.yaml --script scripts/train_avid.py


poetry run python3 -m torch.distributed.launch --nproc_per_node=1 --nnodes=1 \
--master_addr=127.0.0.1 --master_port=12353 --node_rank=0 scripts/train_avid.py \
--base configs/train/avid/avid_145M.yaml --train --devices 1 lightning.trainer.num_nodes=1

